#! /bin/bash

source activate seg && cd /root/code/mmsegmentation
mim train mmseg /root/cityscapes/unet-s5-d16_fcn_4xb4-160k_cityscapes-512x1024.py \
    --amp \
    --cfg-options train_dataloader.batch_size=8 \
    train_dataloader.num_workers=16 \
    optim_wrapper.optimizer.lr=0.005 \
    train_cfg.max_iters=320000 \
    default_hooks.checkpoint.interval=2000 \
    default_hooks.checkpoint.max_keep_ckpts=5 \
    --resume


# 4xb4@lr0.01 i160000
# 1xb16@lr0.01 i160000
# 1xb8@lr0.005 i320000
# 1xb4@lr0.0025 i640000